package com.example.front.component;
import com.example.common.exception.BusinessException;
import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;


/**
 * @author websocket服务
 */
@ServerEndpoint(value = "/voice/{username}")
@Component
public class WebSocketServer {


    /**
     * 记录当前在线连接数
     */
    public static final Map<String, Set<Session>> sessionMap = new ConcurrentHashMap<>();

    /**
     * 连接建立成功调用的方法
     */
    @OnOpen
    public void onOpen(Session session, @PathParam("username") String username) {
        System.out.println("---------------------------------------------->"+username);
        if(sessionMap.get(username)!=null){
            sessionMap.get(username).add(session);
            sessionMap.put(username,sessionMap.get(username));
        }else{
            Set<Session> sessions =new HashSet<>();
            sessions.add(session);
            sessionMap.put(username,sessions);
        }
    }

    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose(Session session, @PathParam("username") String username) {
        Set<Session> sessions=sessionMap.get(username);
        if(sessions.size()==1){
            sessionMap.remove(username);
        } else{
            sessions.remove(session);
            sessionMap.put(username,sessions);
        }
    }

    /**
     * 收到客户端消息后调用的方法
     * 后台收到客户端发送过来的消息
     * onMessage 是一个消息的中转站
     * 接受 浏览器端 socket.send 发送过来的 json数据
     * @param message 客户端发送过来的消息
     */
    @OnMessage
    public void onMessage(String message, Session session, @PathParam("username") String username) {

    }

    @OnError
    public void onError(Session session, Throwable error) {
        new BusinessException("语音播报异常");
        error.printStackTrace();
    }

    /**
     * 服务端发送消息给客户端
     */
    private void sendMessage(String message, Session toSession) {
        try {
            synchronized (toSession) {
                toSession.getBasicRemote().sendText(message);
            }
        } catch (Exception e) {
        }
    }

    /**
     * 服务端发送消息给指定客户端
     */
    public void sendToken(String message,String date) {
        try {
            for (Session session : sessionMap.get(message)) {
                sendMessage(date,session);
            }

        } catch (Exception e) {
        }
    }

    /**
     * 服务端发送消息给所有客户端
     */
    private void sendAllMessage(String message) {

    }
}
